package main

import (
	"gorm.io/driver/sqlite"
	"gorm.io/gen"
	"gorm.io/gorm"
)

func main() {
	// 初始化生成器
	g := gen.NewGenerator(gen.Config{
		OutPath:      "./internal/db/query", // 输出目录
		ModelPkgPath: "./model",
		Mode:         gen.WithDefaultQuery | gen.WithQueryInterface,
	})

	// 连接SQLite数据库
	db, _ := gorm.Open(sqlite.Open("modbus-slave.db"), &gorm.Config{})
	g.UseDB(db) // 使用数据库连接

	// 为所有表生成模型
	g.GenerateAllTable()

	g.ApplyBasic(g.GenerateAllTable()...)

	// 执行生成
	g.Execute()
}
